#pragma once
#include <opencv2/dnn.hpp>
#include<opencv2/opencv.hpp>
using namespace std;
using namespace cv;
class MyModel
{
	dnn::Net net;
public:
	MyModel();
	virtual ~MyModel();

	void loadModel(String modelPath);

	void predict(String imageName);
	void predict(cv::Mat img);

private :
	double* softmax(float  src[], int size, int & maxIndex, double & maxPro);
};

